run_name: triton-matmul-benchmark
image: mosaicml/llm-foundry:2.0.1_cu118-latest
cluster: r9z1
gpu_num: 8
gpu_type: h100_80gb
integrations:
  - integration_type: git_repo
    git_repo: openai/triton
    git_branch: main
    ssh_clone: false
command: |-
  export ENABLE_TMA=1
  export ENABLE_MMA_V3=1
  
  cd triton/python
  pip install . --no-dependencies

  python tutorials/10-experimental-tma-store-matrix-multiplication.py
